import torch.nn as nn
import torch


class Module(nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, input_):
        output = input_ + 1
        return output


a = torch.tensor(1.0)
m = Module()
print(m.forward(a))
